package cn.ichiva.dl4j.gnn.mnist.data;

import java.io.IOException;

import org.nd4j.linalg.dataset.api.iterator.BaseDatasetIterator;

public class PathMnistDataSetIterator extends BaseDatasetIterator {
    public PathMnistDataSetIterator(String mnistRoot, int batch, int numExamples) throws IOException {
        this(mnistRoot,batch, numExamples, false);
    }

    public PathMnistDataSetIterator(String mnistRoot, int batch, int numExamples, boolean binarize) throws IOException {
        this(mnistRoot,batch, numExamples, binarize, true, false, 0L);
    }

    public PathMnistDataSetIterator(String mnistRoot, int batchSize, boolean train, int seed) throws IOException {
        this(mnistRoot,batchSize, train ? '\uea60' : 10000, false, train, true, (long)seed);
    }

    public PathMnistDataSetIterator(String mnistRoot, int batch, int numExamples, boolean binarize, boolean train, boolean shuffle, long rngSeed) throws IOException {
        super(batch, numExamples, new PathMnistDataFetcher(mnistRoot,binarize, train, shuffle, rngSeed));
    }
}
